package com.swotxu.mall.order.conf;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @Date: 2022/4/29 16:06
 * @Author: swotXu
 */
@Slf4j
@Configuration
public class MyRabbitMqConfig {

    @Autowired
    RabbitTemplate rabbitTemplate;

    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制 rabbitTemplate
     * 发送端消息确认
     *  1、注册C2B回调（发送端抵达服务器确认）
     *  2、注册E2Q回调（发送端抵达队列确认）publisher-returns
     * 消费端消息确认
     *  默认是自动确认的，只要收到消息，客户端就会自动确认，broker就会移除这个消息
     *  配置设置：spring.rabbitmq.listener.simple.acknowledge-mode = manual #开启手动ack模式
     *  1.手动签收
     *      channel.basicAck(deliveryTag, false);
     *  2.手动拒签
     *      channel.basicNack(deliveryTag, false, true);
     */
    @PostConstruct
    public void initRabbitTemplate() {
        // 1.设置发送端抵达服务器确认
        rabbitTemplate.setConfirmCallback(this::confirmCallback);
        // 2.发送端抵达队列确认
        rabbitTemplate.setReturnCallback(this::returnedMessage);
    }

    /**
     * 1. 服务器收到消息的回调（消息投递的回调） C(client) -> B(Broker)
     * 注意：配置需开启发送确认
     *      spring.rabbitmq.publisher-confirms=true
     *
     * @param data 当前消息的唯一关联数据（消息唯一id）
     * @param ack   消息是否成功收到 (只要消息抵达Broker，就是true)
     * @param cause 失败原因
     */
    public void confirmCallback(CorrelationData data, boolean ack, String cause) {
        log.info("CorrelationData: {}, ack: {}, cause: {}", data, ack, cause);
    }

    /**
     * 2. 消息抵达队列的回调  E(exchange) -> Q(queue)
     * 注意：配置需开启发送确认
     *      spring.rabbitmq.publisher-returns=true
     *      spring.rabbitmq.template.mandatory=true # 异步优先发送 returnsconfirm 确认
     * 触发时机：只要消息没有投递给指定队列，就触发这个回调
     * @param message       投递失败的消息
     * @param replyCode     回复的状态码
     * @param replyText     回复的内容
     * @param exchange      失败消息的交换机
     * @param routingKey    对应的路由key
     */
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.info("Fail Message: {}, replyCode: {}, replyText: {}, exchange: {}, routingKey: {}"
                , message, replyCode, replyText, exchange, routingKey);
    }
}
